import torch
MLP_model = torch.load('models/MLP.pth', weights_only=True)
print(MLP_model)